package com.librelato.tela.datamodel;

import java.util.List;

import javax.faces.model.ListDataModel;

import org.primefaces.model.SelectableDataModel;

import com.librelato.bean.TransportadoraBean;

public class TransportadoraDataModel extends ListDataModel<TransportadoraBean>
		implements SelectableDataModel<TransportadoraBean> {

	public TransportadoraDataModel(List<TransportadoraBean> data) {
		super(data);
	}

	@SuppressWarnings("unchecked")
	@Override
	public TransportadoraBean getRowData(String linha) {
		List<TransportadoraBean> listaTran = (List<TransportadoraBean>) getWrappedData();
		for (TransportadoraBean transportadora : listaTran) {
			if (transportadora.getCodigo().equals(linha))
				return transportadora;

		}

		return null;
	}

	@Override
	public Object getRowKey(TransportadoraBean linha) {
		return linha.getCodigo();
	}

}
